home *** CD-ROM | disk | FTP | other *** search
/ SGI Freeware 2002 November / SGI Freeware 2002 November - Disc 3.iso / dist / fw_qt3.idb / usr / freeware / Qt / examples / demo / opengl / glgear.cpp.z / glgear.cpp
C/C++ Source or Header  |  2002-04-08  |  7KB  |  259 lines
  1. /****************************************************************************
  2. ** $Id:  qt/glgear.cpp   3.0.3   edited Oct 12 12:18 $
  3. **
  4. ** Copyright (C) 1992-2000 Trolltech AS.  All rights reserved.
  5. **
  6. ** This file is part of an example program for Qt.  This example
  7. ** program may be used, distributed and modified without limitation.
  8. **
  9. *****************************************************************************/
  10. //
  11. // Draws a gear.
  12. //
  13. // Portions of this code have been borrowed from Brian Paul's Mesa
  14. // distribution.
  15. //
  16.  
  17. #include "glgear.h"
  18.  
  19. #include <math.h>
  20.  
  21. #if defined(Q_CC_MSVC)
  22. #pragma warning(disable:4305) // init: truncation from const double to float
  23. #endif
  24.  
  25. /*
  26.  * Draw a gear wheel.  You'll probably want to call this function when
  27.  * building a display list since we do a lot of trig here.
  28.  *
  29.  * Input:  inner_radius - radius of hole at center
  30.  *       outer_radius - radius at center of teeth
  31.  *       width - width of gear
  32.  *       teeth - number of teeth
  33.  *       tooth_depth - depth of tooth
  34.  */
  35. static void gear( GLfloat inner_radius, GLfloat outer_radius, GLfloat width,
  36.           GLint teeth, GLfloat tooth_depth )
  37. {
  38.     GLint i;
  39.     GLfloat r0, r1, r2;
  40.     GLfloat angle, da;
  41.     GLfloat u, v, len;
  42.  
  43.     r0 = inner_radius;
  44.     r1 = outer_radius - tooth_depth/2.0;
  45.     r2 = outer_radius + tooth_depth/2.0;
  46.  
  47.     const double pi = 3.14159264;
  48.     da = 2.0*pi / teeth / 4.0;
  49.  
  50.     glShadeModel( GL_FLAT );
  51.  
  52.     glNormal3f( 0.0, 0.0, 1.0 );
  53.  
  54.     /* draw front face */
  55.     glBegin( GL_QUAD_STRIP );
  56.     for (i=0;i<=teeth;i++) {
  57.     angle = i * 2.0*pi / teeth;
  58.     glVertex3f( r0*cos(angle), r0*sin(angle), width*0.5 );
  59.     glVertex3f( r1*cos(angle), r1*sin(angle), width*0.5 );
  60.     glVertex3f( r0*cos(angle), r0*sin(angle), width*0.5 );
  61.     glVertex3f( r1*cos(angle+3*da), r1*sin(angle+3*da), width*0.5 );
  62.     }
  63.     glEnd();
  64.  
  65.     /* draw front sides of teeth */
  66.     glBegin( GL_QUADS );
  67.     da = 2.0*pi / teeth / 4.0;
  68.     for (i=0;i<teeth;i++) {
  69.     angle = i * 2.0*pi / teeth;
  70.  
  71.     glVertex3f( r1*cos(angle),      r1*sin(angle),      width*0.5 );
  72.     glVertex3f( r2*cos(angle+da),   r2*sin(angle+da),      width*0.5 );
  73.     glVertex3f( r2*cos(angle+2*da), r2*sin(angle+2*da), width*0.5 );
  74.     glVertex3f( r1*cos(angle+3*da), r1*sin(angle+3*da), width*0.5 );
  75.     }
  76.     glEnd();
  77.  
  78.  
  79.     glNormal3f( 0.0, 0.0, -1.0 );
  80.  
  81.     /* draw back face */
  82.     glBegin( GL_QUAD_STRIP );
  83.     for (i=0;i<=teeth;i++) {
  84.     angle = i * 2.0*pi / teeth;
  85.     glVertex3f( r1*cos(angle), r1*sin(angle), -width*0.5 );
  86.     glVertex3f( r0*cos(angle), r0*sin(angle), -width*0.5 );
  87.     glVertex3f( r1*cos(angle+3*da), r1*sin(angle+3*da), -width*0.5 );
  88.     glVertex3f( r0*cos(angle), r0*sin(angle), -width*0.5 );
  89.     }
  90.     glEnd();
  91.  
  92.     /* draw back sides of teeth */
  93.     glBegin( GL_QUADS );
  94.     da = 2.0*pi / teeth / 4.0;
  95.     for (i=0;i<teeth;i++) {
  96.     angle = i * 2.0*pi / teeth;
  97.  
  98.     glVertex3f( r1*cos(angle+3*da), r1*sin(angle+3*da), -width*0.5 );
  99.     glVertex3f( r2*cos(angle+2*da), r2*sin(angle+2*da), -width*0.5 );
  100.     glVertex3f( r2*cos(angle+da),   r2*sin(angle+da),      -width*0.5 );
  101.     glVertex3f( r1*cos(angle),      r1*sin(angle),      -width*0.5 );
  102.     }
  103.     glEnd();
  104.  
  105.  
  106.     /* draw outward faces of teeth */
  107.     glBegin( GL_QUAD_STRIP );
  108.     for (i=0;i<teeth;i++) {
  109.     angle = i * 2.0*pi / teeth;
  110.  
  111.     glVertex3f( r1*cos(angle),      r1*sin(angle),       width*0.5 );
  112.     glVertex3f( r1*cos(angle),      r1*sin(angle),      -width*0.5 );
  113.     u = r2*cos(angle+da) - r1*cos(angle);
  114.     v = r2*sin(angle+da) - r1*sin(angle);
  115.     len = sqrt( u*u + v*v );
  116.     u /= len;
  117.     v /= len;
  118.     glNormal3f( v, -u, 0.0 );
  119.     glVertex3f( r2*cos(angle+da),   r2*sin(angle+da),       width*0.5 );
  120.     glVertex3f( r2*cos(angle+da),   r2*sin(angle+da),      -width*0.5 );
  121.     glNormal3f( cos(angle), sin(angle), 0.0 );
  122.     glVertex3f( r2*cos(angle+2*da), r2*sin(angle+2*da),  width*0.5 );
  123.     glVertex3f( r2*cos(angle+2*da), r2*sin(angle+2*da), -width*0.5 );
  124.     u = r1*cos(angle+3*da) - r2*cos(angle+2*da);
  125.     v = r1*sin(angle+3*da) - r2*sin(angle+2*da);
  126.     glNormal3f( v, -u, 0.0 );
  127.     glVertex3f( r1*cos(angle+3*da), r1*sin(angle+3*da),  width*0.5 );
  128.     glVertex3f( r1*cos(angle+3*da), r1*sin(angle+3*da), -width*0.5 );
  129.     glNormal3f( cos(angle), sin(angle), 0.0 );
  130.     }
  131.  
  132.     glVertex3f( r1*cos(0.0), r1*sin(0.0), width*0.5 );
  133.     glVertex3f( r1*cos(0.0), r1*sin(0.0), -width*0.5 );
  134.  
  135.     glEnd();
  136.  
  137.  
  138.     glShadeModel( GL_SMOOTH );
  139.  
  140.     /* draw inside radius cylinder */
  141.     glBegin( GL_QUAD_STRIP );
  142.     for (i=0;i<=teeth;i++) {
  143.     angle = i * 2.0*pi / teeth;
  144.     glNormal3f( -cos(angle), -sin(angle), 0.0 );
  145.     glVertex3f( r0*cos(angle), r0*sin(angle), -width*0.5 );
  146.     glVertex3f( r0*cos(angle), r0*sin(angle), width*0.5 );
  147.     }
  148.     glEnd();
  149.  
  150. }
  151.  
  152. void GLGear::draw()
  153. {
  154.     glClear( GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT );
  155.  
  156.     glPushMatrix();
  157.     glRotatef( view_rotx, 1.0, 0.0, 0.0 );
  158.     glRotatef( view_roty, 0.0, 1.0, 0.0 );
  159.     glRotatef( view_rotz, 0.0, 0.0, 1.0 );
  160.  
  161.     glPushMatrix();
  162.     glTranslatef( -3.0, -2.0, 0.0 );
  163.     glRotatef( angle, 0.0, 0.0, 1.0 );
  164.     glCallList(gear1);
  165.     glPopMatrix();
  166.  
  167.     glPushMatrix();
  168.     glTranslatef( 3.1, -2.0, 0.0 );
  169.     glRotatef( -2.0*angle-9.0, 0.0, 0.0, 1.0 );
  170.     glCallList(gear2);
  171.     glPopMatrix();
  172.  
  173.     glPushMatrix();
  174.     glTranslatef( -3.1, 2.2, -1.8 );
  175.     glRotatef( 90.0, 1.0, 0.0, 0.0 );
  176.     glRotatef( 2.0*angle-2.0, 0.0, 0.0, 1.0 );
  177.     glCallList(gear3);
  178.     glPopMatrix();
  179.  
  180.     glPopMatrix();
  181. }
  182.  
  183. GLGear::GLGear( QWidget *parent, const char *name, WFlags f )
  184.      : GLControlWidget( parent, name, 0, f )
  185. {
  186.     scale = 1.0;
  187.     setAnimationDelay( 15 );
  188.     view_rotx = 20.0;
  189.     view_roty = 30.0;
  190.     view_rotz = 0.0;
  191.     angle = 0.0;
  192. }
  193.  
  194. void GLGear::initializeGL()
  195. {
  196.     static GLfloat pos[4] = {5.0, 5.0, 10.0, 1.0 };
  197.     static GLfloat ared[4] = {0.8, 0.1, 0.0, 1.0 };
  198.     static GLfloat agreen[4] = {0.0, 0.8, 0.2, 1.0 };
  199.     static GLfloat ablue[4] = {0.2, 0.2, 1.0, 1.0 };
  200.  
  201.     glLightfv( GL_LIGHT0, GL_POSITION, pos );
  202.     glEnable( GL_CULL_FACE );
  203.     glEnable( GL_LIGHTING );
  204.     glEnable( GL_LIGHT0 );
  205.     glEnable( GL_DEPTH_TEST );
  206.  
  207.     /* make the gears */
  208.     gear1 = glGenLists(1);
  209.     glNewList(gear1, GL_COMPILE);
  210.     glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, ared );
  211.     gear( 1.0, 4.0, 1.0, 20, 0.7 );
  212.     glEndList();
  213.  
  214.     gear2 = glGenLists(1);
  215.     glNewList(gear2, GL_COMPILE);
  216.     glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, agreen );
  217.     gear( 0.5, 2.0, 2.0, 10, 0.7 );
  218.     glEndList();
  219.  
  220.     gear3 = glGenLists(1);
  221.     glNewList(gear3, GL_COMPILE);
  222.     glMaterialfv( GL_FRONT, GL_AMBIENT_AND_DIFFUSE, ablue );
  223.     gear( 1.3, 2.0, 0.5, 10, 0.7 );
  224.     glEndList();
  225.  
  226.     glEnable( GL_NORMALIZE );
  227. }
  228.  
  229.  
  230. void GLGear::resizeGL( int width, int height )
  231. {
  232.     GLfloat w = (float) width / (float) height;
  233.     GLfloat h = 1.0;
  234.  
  235.     glViewport( 0, 0, width, height );
  236.     glMatrixMode(GL_PROJECTION);
  237.     glLoadIdentity();
  238.     glFrustum( -w, w, -h, h, 5.0, 60.0 );
  239.     glMatrixMode(GL_MODELVIEW);
  240.     glLoadIdentity();
  241.     glTranslatef( 0.0, 0.0, -40.0 );
  242. }
  243.  
  244.  
  245. void GLGear::paintGL()
  246. {
  247.     glPushMatrix();
  248.     transform();
  249.     draw();
  250.     glPopMatrix();
  251. }
  252.  
  253. void GLGear::animate()
  254. {
  255.     angle += 2.0;
  256.     view_roty += 1.0;
  257.     updateGL();
  258. }
  259.